from openai import OpenAI

# 解决请求超时问题
import os
os.environ["http_proxy"] = "http://localhost:7890"
os.environ["https_proxy"] = "http://localhost:7890"

# 需要在实例化OpenAI对象时候，指定秘钥
client = OpenAI(api_key="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")

completion = client.chat.completions.create(
  model="gpt-3.5-turbo", # 指定使用的大模型是谁，gpt-3.5-turbo 是chatgpt3.5，
  messages=[  # 提示词，
    # {"role": "身份", "content": "你是一个老师."},  # 让chatgpt充当什么身份
    # {"role": "user", "content": "请帮我基于中文."},  # 约束条件或基于什么条件
    {"role": "assistant", "content": "编写一段歌颂元宵节的歌词"} # 具体要chatgpt完成的操作了。
  ]
)

print(completion.choices[0].message)